package com.ruoyi.system.controller;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Iterator;
import java.util.List;

import com.ruoyi.common.redis.service.RedisService;
import com.ruoyi.system.dto.UpUser;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.CollectionUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.core.utils.file.FileTypeUtils;
import com.ruoyi.common.core.utils.file.MimeTypeUtils;
import com.ruoyi.common.core.web.controller.BaseController;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.common.log.annotation.Log;
import com.ruoyi.common.log.enums.BusinessType;
import com.ruoyi.common.security.service.TokenService;
import com.ruoyi.common.security.utils.SecurityUtils;
import com.ruoyi.system.api.RemoteFileService;
import com.ruoyi.system.api.domain.SysFile;
import com.ruoyi.system.api.domain.SysUser;
import com.ruoyi.system.api.model.LoginUser;
import com.ruoyi.system.service.ISysUserService;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;

import javax.servlet.http.HttpServletRequest;

/**
 * 个人信息 业务处理
 * 
 * @author ruoyi
 */
@RestController
@RequestMapping("/user/profile")
public class SysProfileController extends BaseController
{
    @Autowired
    private ISysUserService userService;
    
    @Autowired
    private TokenService tokenService;
    
    @Autowired
    private RemoteFileService remoteFileService;

    /**
     * 个人信息
     */
    @GetMapping
    public AjaxResult profile()
    {
        String username = SecurityUtils.getUsername();
        SysUser user = userService.selectUserByUserName(username);
        AjaxResult ajax = AjaxResult.success(user);
//        ajax.put("roleGroup", userService.selectUserRoleGroup(username));
//        ajax.put("postGroup", userService.selectUserPostGroup(username));
        return ajax;
    }

    /**
     * 修改用户
     */
    @Log(title = "个人信息", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult updateProfile(@RequestBody SysUser user)
    {
        LoginUser loginUser = SecurityUtils.getLoginUser();
        SysUser sysUser = loginUser.getSysUser();
        user.setUserName(sysUser.getUserName());
        if (StringUtils.isNotEmpty(user.getPhonenumber()) && !userService.checkPhoneUnique(user))
        {
            return error("修改用户'" + user.getUserName() + "'失败，手机号码已存在");
        }
        else if (StringUtils.isNotEmpty(user.getEmail()) && !userService.checkEmailUnique(user))
        {
            return error("修改用户'" + user.getUserName() + "'失败，邮箱账号已存在");
        }
        user.setUserId(sysUser.getUserId());
        user.setPassword(null);
        user.setAvatar(null);
        user.setDeptId(null);
        if (userService.updateUserProfile(user) > 0)
        {
            // 更新缓存用户信息
            loginUser.getSysUser().setNickName(user.getNickName());
            loginUser.getSysUser().setPhonenumber(user.getPhonenumber());
            loginUser.getSysUser().setEmail(user.getEmail());
            loginUser.getSysUser().setSex(user.getSex());
            tokenService.setLoginUser(loginUser);
            return success();
        }
        return error("修改个人信息异常，请联系管理员");
    }

    /**
     * 重置密码
     */
    @Log(title = "个人信息", businessType = BusinessType.UPDATE)
    @PutMapping("/updatePwd")
    public AjaxResult updatePwd(@RequestBody UpUser upUser)
    {
        String oldPassword = upUser.getOldPassword();
        String newPassword = upUser.getNewPassword();
        String username = SecurityUtils.getUsername();
        SysUser user = userService.selectUserByUserName(username);
        String password = user.getPassword();
        if (!SecurityUtils.matchesPassword(oldPassword, password))
        {
            return error("修改密码失败，旧密码错误");
        }
        if (SecurityUtils.matchesPassword(newPassword, password))
        {
            return error("新密码不能与旧密码相同");
        }
        if (userService.resetUserPwd(username, SecurityUtils.encryptPassword(newPassword)) > 0)
        {
            // 更新缓存用户密码
            LoginUser loginUser = SecurityUtils.getLoginUser();
            loginUser.getSysUser().setPassword(SecurityUtils.encryptPassword(newPassword));
            tokenService.setLoginUser(loginUser);
            return success();
        }
        return error("修改密码异常，请联系管理员");
    }

    @Autowired
    RedisService redisService;

    /**
     * 忘记密码
     */
    @Log(title = "个人信息", businessType = BusinessType.UPDATE)
    @PutMapping("/updatePwdByPhone")
    public AjaxResult updatePwdByPhone(@RequestBody UpUser upUser)
    {
        String phone = upUser.getPhone();
        String username = upUser.getPhone();
        String code = upUser.getCode();
        String newPassword = upUser.getNewPassword();

        SysUser sysUser = userService.selectUserByUserPhone(phone);
        if (sysUser == null){
            return error("手机号码不存在！");
        }
        String redisCode = redisService.getCacheObject("verificationCode" + phone);
        if(org.apache.commons.lang3.StringUtils.isEmpty(redisCode)){
            return error("验证码失效,请再次发送!");
        }
        if(!code.equals(redisCode)){
            return error("验证码错误");
        }
        redisService.deleteObject("verificationCode" + phone);
        userService.resetUserPwd(username, SecurityUtils.encryptPassword(newPassword));
        return success();
    }

    /**
     * 更新手机号码
     */
    @Log(title = "个人信息", businessType = BusinessType.UPDATE)
    @PutMapping("/updatePhone")
    public AjaxResult updatePhone(@RequestBody UpUser upUser)
    {
        String username = SecurityUtils.getUsername();
        SysUser user = userService.selectUserByUserName(username);
        String password = user.getPassword();
        String code = upUser.getCode();
        String phone = upUser.getPhone();

        if (!SecurityUtils.matchesPassword(upUser.getPassword(), password))
        {
            return error("密码错误!");
        }

        String redisCode = redisService.getCacheObject("verificationCode" + phone);
        if(StringUtils.isEmpty(redisCode)){
            return error("验证码失效,请再次发送!");
        }
        if(!code.equals(redisCode)){
            return error("验证码错误");
        }

        user.setPhonenumber(phone);
        userService.updateUserProfile(user);

        redisService.deleteObject("verificationCode" + phone);

        return success();
    }


    /**
     * 头像上传
     */
    @Log(title = "用户头像", businessType = BusinessType.UPDATE)
    @PostMapping("/avatar")
    public AjaxResult avatar(@RequestParam("avatarfile") MultipartFile file)
    {
        if (!file.isEmpty())
        {
            LoginUser loginUser = SecurityUtils.getLoginUser();
            String extension = FileTypeUtils.getExtension(file);
            if (!StringUtils.equalsAnyIgnoreCase(extension, MimeTypeUtils.IMAGE_EXTENSION))
            {
                return error("文件格式不正确，请上传" + Arrays.toString(MimeTypeUtils.IMAGE_EXTENSION) + "格式");
            }
            R<SysFile> fileResult = remoteFileService.upload(file);
            if (StringUtils.isNull(fileResult) || StringUtils.isNull(fileResult.getData()))
            {
                return error("文件服务异常，请联系管理员");
            }
            String url = fileResult.getData().getUrl();
            if (userService.updateUserAvatar(loginUser.getUsername(), url))
            {
                AjaxResult ajax = AjaxResult.success();
                ajax.put("imgUrl", url);
                // 更新缓存用户头像
                loginUser.getSysUser().setAvatar(url);
                tokenService.setLoginUser(loginUser);
                return ajax;
            }
        }
        return error("上传图片异常，请联系管理员");
    }


    /**
     * 图片上传
     *
     * @return
     */
    @RequestMapping("/upLoadFile")
    public AjaxResult upLoadFiles(HttpServletRequest request) throws Exception {
        String fileUrl = "";
        int index = 0 ;
        FileOutputStream fileOutputStream = null;
        FileInputStream fileInputStream = null;
        String url = "";
        try {

            List<MultipartFile> multipartFiles = extractedFiles(request);
            if (CollectionUtils.isEmpty(multipartFiles)) {
                return error("没有检测到文件存在");
            }

            for (MultipartFile multipartFile:multipartFiles ) {
                index = multipartFile.getOriginalFilename().lastIndexOf('.') + 1;
                String fileType = multipartFile.getOriginalFilename().substring(index);
                url = multipartFile.getOriginalFilename();
                logger.warn("================fileType:",fileType);
//                // 组织文件名
//                if (fileType != null && !fileType.equals("") && !allowedSuffix.contains(fileType.toLowerCase())) {
//                    return error("非法的文件后缀");
//                }
//                File folder = new File(url);
//                if (!folder.exists()) {
//                    folder.mkdir();
//                }
//
//                fileUrl = UUIDUtils.getUuid32() + "." + fileType;
//                String fileStr = url + "/" + fileUrl;
            }
        }catch (Exception ex){
            logger.error(ex.getMessage(),ex);
        }
        return success(url);
    }
    /**
     * 提取文件
     *
     * @param request
     * @return
     */
    private List<MultipartFile> extractedFiles(HttpServletRequest request) {
        List<MultipartFile> multipartFiles = new ArrayList<>();
        // 创建一个通用的多部分解析器
        CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(
                request.getSession().getServletContext());
        // 判断 request 是否有文件上传,即多部分请求
        if (multipartResolver.isMultipart(request)) {
            // 转换成多部分request
            MultipartHttpServletRequest multiRequest = (MultipartHttpServletRequest) request;
            // 取得request中的所有文件名
            Iterator<String> iter = multiRequest.getFileNames();
            while (iter.hasNext()) {
                // 记录上传过程起始时的时间，用来计算上传时间
                long pre = System.currentTimeMillis();
                // 取得上传文件
                MultipartFile file = multiRequest.getFile(iter.next());
                if (file != null) {
                    multipartFiles.add(file);
                    // 记录上传该文件后的时间
                    long finaltime = System.currentTimeMillis();
                }
            }
        }
        return multipartFiles;
    }

}
